import torch
from torchvision.models import *
import numpy as np
from data_class import *
import torchvision
from torch import nn
